# -*- encoding: utf-8 -*-
"""
========================================
@Time   ：2021/9/6 12:08
@Auther ：shutao
@File   ：structure_of_hamiltonian.py
@IDE    ：PyCharm
@Github ：https://github.com/NameLacker
@Gitee  ：https://gitee.com/nameLacker
========================================
"""

from paddle_quantum.qchem import geometry, get_molecular_data

import warnings

warnings.filterwarnings("ignore")

# in Angstrom
h2o_structure_direct = [["H", [-0.02111417, 0.8350417, 1.47688078]],  # H 代表着水分子中的氢元素
                        ["O", [0.0, 0.0, 0.0]],  # O 代表着水分子中的氧元素
                        ["H", [-0.00201087, 0.45191737, -0.27300254]]]

h2o_structure_xyz = geometry(file="h2o.xyz")
assert h2o_structure_xyz == h2o_structure_direct

h2o_moledata = get_molecular_data(
    h2o_structure_direct,
    charge=0,                # 水分子是中性的，不带电
    multiplicity=1,          # 水分子只有一个未配对电子
    basis="sto-3g",
    method="scf",
    if_save=True,            # 是否将 MolecularData 中的信息存储成 hdf5 文件
    if_print=True,           # 是否需要打印出水分子的基态能量
    name="",                 # 指定 hdf5 文件的名字
    file_path="."            # 指定 hdf5 文件的路径
)
